tensor([[[[[0.7500, 0.1600],
           [0.1600, 0.7500]],

          [[0.1400, 0.2400],
           [0.2400, 0.1400]]],


         [[[0.1400, 0.2400],
           [0.2400, 0.1400]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.2900],
           [0.2900, 0.0000]],

          [[0.5100, 0.6500],
           [0.6500, 0.5100]]],


         [[[0.5100, 0.6500],
           [0.6500, 0.5100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.2500, 0.5400],
           [0.5400, 0.2500]],

          [[0.3500, 0.1100],
           [0.1100, 0.3500]]],


         [[[0.3500, 0.1100],
           [0.1100, 0.3500]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)